import kNN
import numpy as np

def createDataSet():
    groups = np.array([[3,104],[2,100],[1,81],[101,10],[99,5],[98,2]])
    labels = ['爱情片','爱情片','爱情片','动作片','动作片','动作片']
    return groups, labels

groups,labels=createDataSet()

result =kNN.classify0([18,90],groups,labels,3)

